UAVDT目标检测数据集格式转化以及训练集验证集划分教程(原格式转化为Json(COCO)格式) |
您所在的位置:网站首页 › 无人机 目标检测 › UAVDT目标检测数据集格式转化以及训练集验证集划分教程(原格式转化为Json(COCO)格式) |
UAVDT目标检测数据集格式转化教程(原格式转化为Json(COCO)格式)
无人机目标检测数据集目前主要有两个,一个是VisDrone19/18,还有一个就是UAVDT。 UAVDT数据集不是COCO格式的,但是目标检测一般使用COCO或者VOC格式的。 我找了一圈,博客和Github上都没有相关的转化数据集代码。(以前转数据集网上一搜都有别人提供的,但是这个数据集可能用的人少吧,我没找到)。所以只能自己来写了。以后其他人用这个数据集需要转化格式的时候就可以参考了。 1、下载数据集,了解数据集格式。 官网是这个:UAVDT官网 里面的图片类似这样。 下载下来后我们主要用这两个文件。 2、先处理图片,把图片重命名放入到一个文件夹中。 COCO格式中图片都是放入到单独一个文件夹中的。 这个数据集一个文件夹下又有好多文件夹,这样不太好,我们可以把图片重命名后放入一个文件夹。 这里可以写一个简单的Python脚本。就叫imagescopy2onedir.py吧 import os import shutil multidirpath = 'D:/07_codeD/datasets/UAV/UAV-benchmark-M' outdir = 'D:/07_codeD/datasets/UAV/uavdtallimages_rename' # os.mkdir((outdir)) filenames = os.listdir(multidirpath) # print(filenames) for file in filenames: wholefile = multidirpath + '/' + file filenextname = os.listdir(wholefile) # print(filenextname) for filenext in filenextname: pathall = multidirpath + '/' + file + '/' + filenext # print(pathall) str1 = str(file) outpath = outdir + '/' + str1 + '_' + filenext[-10:] # print(outpath) shutil.copyfile(pathall,outpath)改变路径就可以在你自己的电脑上运行了。 这是我们重新命名的图片。 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |